# yolo_config.py

YOLO_IC_CONFIG = {
    'model_path': 'yolo_IC/best_epoch_weights.pth',
    'classes_path': 'yolo_IC/class.txt',
    'input_shape': [160, 160],
    'phi': 'l',
    'confidence': 0.1,
    'nms_iou': 0.3,
    'letterbox_image': True,
    'cuda': True
}

YOLO_CN_IC_CONFIG = {
    'model_path': './AI/weights/small_ng_weights.pth',
    'classes_path': 'AI/classes/class.txt',
    'input_shape': [320, 320],
    'phi': 'l',
    'confidence': 0.01,
    'nms_iou': 0.3,
    'letterbox_image': True,
    'cuda': True
}